import MySQLdb as mysql

no = int(input('编号: '))
name = input('名称: ')
location = input('所在地: ')
# 第1步：创建MySQL数据库连接，获取连接对象
conn = mysql.connect(host='47.104.31.138', port=3306,
                     user='root', password='Luohao.618',
                     database='hrs', charset='utf8mb4',
                     connect_timeout=3)
try:
    # 第2步：基于连接对象创建游标对象
    with conn.cursor() as cursor:
        # 第3步：通过游标对象向数据库发出SQL语句并获取执行结果
        affected_rows = cursor.execute(
            'insert into tb_dept values (%s, %s, %s)',
            (no, name, location)
        )
        if affected_rows == 1:
            print('新增部门成功!!!')
    # 第4步：操作成功：通过连接对象执行提交操作（让之前的操作生效）
    conn.commit()
except mysql.MySQLError as err:
    print(err)
    # 第4步：操作失败：通过连接对象执行回滚操作（撤销之前的操作）
    conn.rollback()
finally:
    # 第5步：关闭连接，释放资源
    conn.close()
